import sys
import os

addr_to_zh = {'AL': '阿拉巴马州', 'AK': '阿拉斯加州', 'AZ': '亚利桑那州', 'AR': '阿肯色州', 'CA': '加利福尼亚州',
              'CO': '科罗拉多州', 'CT': '康涅狄格州', 'DE': '特拉华州', 'FL': '佛罗里达州', 'GA': '乔治亚州',
              'HI': '夏威夷州', 'ID': '爱达荷州', 'IL': '伊利诺斯州', 'IN': '印第安纳州', 'IA': '爱荷华州',
              'KS': '堪萨斯州', 'KY': '肯塔基州', 'LA': '路易斯安那州', 'ME': '缅因州', 'MD': '马里兰州',
              'MA': '马萨诸塞州', 'MI': '密歇根州', 'MN': '明尼苏达州', 'MS': '密西西比州', 'MO': '密苏里州',
              'MT': '蒙大拿州', 'NE': '内布拉斯加州', 'NV': '内华达州', 'NH': '新罕布什尔州', 'NJ': '新泽西州',
              'NM': '新墨西哥州', 'NY': '纽约州', 'NC': '北卡罗来纳州', 'ND': '北达科他州', 'OH': '俄亥俄州',
              'OK': '俄克拉荷马州', 'OR': '俄勒冈州', 'PA': '宾夕法尼亚州', 'RI': '罗得岛州', 'SC': '南卡罗来纳州',
              'SD': '南达科他州', 'TN': '田纳西州', 'TX': '得克萨斯州', 'UT': '犹他州', 'VT': '佛蒙特州',
              'VA': '弗吉尼亚州', 'WA': '华盛顿州', 'WV': '西弗吉尼亚州', 'WI': '威斯康辛州', 'WY': '怀俄明州',
              'PR': '波多黎各自治邦', 'DC': '华盛顿哥伦比亚特区'}

addr_to_en = {'AL': 'Alabama', 'AK': 'Alaska', 'AZ': 'Arizona', 'AR': 'Arkansas', 'CA': 'California',
              'CO': 'Colorado', 'CT': 'Connecticut', 'DE': 'Delaware', 'FL': 'Florida', 'GA': 'Georgia',
              'HI': 'Hawaii', 'ID': 'Idaho', 'IL': 'Illinois', 'IN': 'Indiana', 'IA': 'Iowa',
              'KS': 'Kansas', 'KY': 'Kentucky', 'LA': 'Louisiana', 'ME': 'Maine', 'MD': 'Maryland',
              'MA': 'Massachusetts', 'MI': 'Michigan', 'MN': 'Minnesota', 'MS': 'Mississippi',
              'MO': 'Missouri',
              'MT': 'Montana', 'NE': 'Nebraska', 'NV': 'Nevada', 'NH': 'New Hampshire', 'NJ': 'New Jersey',
              'NM': 'New Mexico', 'NY': 'New York', 'NC': 'North Carolina', 'ND': 'North Dakota',
              'OH': 'Ohio',
              'OK': 'Oklahoma', 'OR': 'Oregon', 'PA': 'Pennsylvania', 'RI': 'Rhode Island',
              'SC': 'South Carolina',
              'SD': 'South Dakota', 'TN': 'Tennessee', 'TX': 'Texas', 'UT': 'Utah', 'VT': 'Vermont',
              'VA': 'Virginia', 'WA': 'Washington', 'WV': 'West Virginia', 'WI': 'Wisconsin',
              'WY': 'Wyoming',
              'PR': 'Puerto Rico', 'DC': 'District of Columbia'}

en_to_abbr = dict(zip(addr_to_en.values(), addr_to_en.keys()))

zh_to_abbr = dict(zip(addr_to_zh.values(), addr_to_zh.keys()))

en_to_zh = dict(zip(addr_to_en.values(), addr_to_zh.values()))

zh_to_en = dict(zip(addr_to_zh.values(), addr_to_en.values()))


def rename_geojson(path):
    file_list = os.listdir(path)
    for filename in file_list:
        name = filename.split(".")[0]
        if addr_to_zh.get(name):
            new_filename = addr_to_zh[name] + ".json"
            os.rename(path + '/' + filename, path + '/' + new_filename)


if __name__ == "__main__":
    rename_geojson(sys.argv[1])
